"""
数据库初始化脚本
运行此脚本创建数据库表
"""
from database import engine, Base
from logger import logging
# 导入所有模型以便 SQLAlchemy 可以创建对应的表
from models import User, SystemConfig, Conversation, Question


def init_database():
    """初始化数据库表"""
    try:
        logging.info("开始创建数据库表...")
        Base.metadata.create_all(bind=engine)
        logging.info("数据库表创建成功!")
    except Exception as e:
        logging.error(f"数据库表创建失败: {str(e)}")
        raise


if __name__ == "__main__":
    init_database()
